MybatisPlus之新增操作并返回主键ID |
您所在的位置：网站首页 › mybatis 批量新增如何将主键返回 › MybatisPlus之新增操作并返回主键ID |
MybatisPlus之新增操作并返回主键ID
|
在应用mybatisplus持久层框架的项目中，经常遇到执行新增操作后需要获取主键ID的场景，下面将分析及测试过程记录分享出来。 1、MybatisPlus新增方法
持久层新增方法源码如下： public interface BaseMapper extends Mapper { int insert(T entity); ... }业务层新增方法源码如下： public interface IService { int DEFAULT_BATCH_SIZE = 1000; default boolean save(T entity) { return SqlHelper.retBool(this.getBaseMapper().insert(entity)); } @Transactional( rollbackFor = {Exception.class} ) default boolean saveBatch(Collection entityList) { return this.saveBatch(entityList, 1000); } boolean saveBatch(Collection entityList, int batchSize); @Transactional( rollbackFor = {Exception.class} ) default boolean saveOrUpdateBatch(Collection entityList) { return this.saveOrUpdateBatch(entityList, 1000); } boolean saveOrUpdateBatch(Collection entityList, int batchSize); ... } 2、表结构 DROP TABLE IF EXISTS t_user; CREATE TABLE `t_user` ( `id` bigint(20) UNSIGNED NOT NULL AUTO_INCREMENT COMMENT '主键ID', `name` varchar(30) NOT NULL DEFAULT '' COMMENT '姓名', `age` int(11) NULL DEFAULT NULL COMMENT '年龄', `gender` tinyint(2) NOT NULL DEFAULT 0 COMMENT '性别，0：女 1：男', PRIMARY KEY (`id`) ) COMMENT = '用户表'; 3、实体类 @Data @TableName("t_user") public class User { /** * 主键 ID, @TableId 注解定义字段为表的主键，type 表示主键类型，IdType.AUTO 表示随着数据库 ID 自增 */ @TableId(type = IdType.AUTO) private Long id; /** * 姓名 */ private String name; /** * 年龄 */ private Integer age; /** * 性别 */ private Integer gender; }@TableName 表名注解，标识实体类对应的表。 当实体类名称和实际表名一致时，如实体名为 User， 表名为 user ，可不用添加该注解，Mybatis Plus 会自动识别并映射到该表。当实体类名称和实际表名不一致时，如实体名为 User， 表名为 t_user，需手动添加该注解，并填写实际表名称。@TableId 主键注解，声明实体类中的主键对应的字段。
IdType 主键类型
Mybatis Plus 对 Mapper 层和 Service 层都将常见的增删改查操作都封装好了，只需简单的继承，即可轻松搞定对数据的增删改查，此次分析新增数据这块。 4.1、mapper层定义一个 UserMapper , 让其继承 BaseMapper，如下 public interface UserMapper extends BaseMapper { }BaseMapper 提供的新增方法仅一个 insert() 方法，如下 User user = new User(); user.setName("犬小哈"); user.setAge(30); user.setGender(1); userMapper.insert(user); // 获取插入数据的主键 ID Long id = user.getId(); System.out.println("id:" + id); 4.2、service层Mybatis Plus 同样也封装了通用的 Service 层 CRUD 操作，并且提供了更丰富的方法。 public interface UserService extends IService { } @Service public class UserServiceImpl extends ServiceImpl implements UserService { }先定义 UserService 接口 ，让其继承自 IService，再定义实现类 UserServiceImpl，让其继承自 ServiceImpl, 同时实现 UserService 接口，这样就可以让 UserService 拥有了基础通用的 CRUD 功能。 与 Mapper 层不同的是，Service 层的新增方法均以 save 开头，并且功能更丰富，方法如下： // 新增数据 sava(T) : boolean // 伪批量插入，实际上是通过 for 循环一条一条的插入 savaBatch(Collection) : boolean // 伪批量插入，int 表示批量提交数，默认为 1000 savaBatch(Collection, int) : boolean // 新增或更新(单条数据) saveOrUpdate(T) : boolean // 批量新增或更新 saveOrUpdateBatch(Collection) : boolean // 批量新增或更新(可指定批量提交数) saveOrUpdateBatch(Collection, int) : boolean 4.2.1、sava(T) // 新增数据 // 实际执行 SQL : INSERT INTO user ( name, age, gender ) VALUES ( '曹操 ', 30, 1 ) User user = new User(); user.setName("曹操 "); user.setAge(30); user.setGender(1); boolean isSuccess = userService.save(user); // 返回主键ID Long id = user.getId(); System.out.println("isSuccess:" + isSuccess); System.out.println("主键 ID: " + id); 4.2.2、savaBatch(Collection)伪批量插入，注意，命名虽然包含了批量的意思，但这不是真的批量插入，不信的话，可以实际测试一下： // 批量插入 List users = new ArrayList(); for (int i = 0; i < 5; i++) { User user = new User(); user.setName("曹操" + i); user.setAge(i); user.setGender(1); users.add(user); } boolean isSuccess = userService.saveBatch(users); System.out.println("isSuccess:" + isSuccess);Mybatis Plus 这个伪批量插入性能其实好些，内部会将每次的插入语句缓存起来，等到达到 1000 条的时候，才会统一推给数据库，虽然最终在数据库那边还是一条一条的执行 INSERT，但还是在和数据库交互的 IO 上做了优化。 4.2.3、savaBatch(Collection, int)多了个 batchSize 参数，可以手动指定批处理的大小，即多少 SQL 操作执行一次，默认为 1000。 4.2.4、saveOrUpdate(T)保存或者更新。即当你需要执行的数据，数据库中不存在时，就执行插入操作，如下： // 实际执行 SQL : INSERT INTO user ( name, age, gender ) VALUES ( '曹操', 60, 1 ) User user = new User(); user.setName("曹操"); user.setAge(60); user.setGender(1); userService.saveOrUpdate(user);当你需要执行的数据，数据库中已存在时，就执行更新操作。框架是如何判断该记录是否存在呢？ 如设置了主键 ID，因为主键 ID 必须是唯一的，Mybatis Plus 会先执行查询操作，判断数据是否存在，存在即执行更新，否则，执行插入操作，如下： User user = new User(); // 设置了主键字段 user.setId(1L); user.setName("曹操"); user.setAge(60); user.setGender(1); userService.saveOrUpdate(user);若是需要根据其他字段判断数据是否存在，可以通过对选定字段添加唯一索引的方式实现。 通过on udplicate key update 的方式实现批量的插入或更新，这种方式需要有唯一索引，通过唯一索引去判断是否冲突，有冲突就会更新，没有冲突就会插入数据。 4.2.5、saveOrUpdateBatch(Collection)批量保存或者更新，示例代码如下： List users = new ArrayList(); for (int i = 0; i < 5; i++) { User user = new User(); user.setId(Long.valueOf(i)); user.setName("曹操" + i); user.setAge(i+1); user.setGender(1); users.add(user); } userService.saveOrUpdateBatch(users); 4.2.6、saveOrUpdateBatch(Collection, int)批量保存或者更新（可手动指定批量大小），示例代码如下： List users = new ArrayList(); for (int i = 0; i < 5; i++) { User user = new User(); user.setId(Long.valueOf(i)); user.setName("曹操" + i); user.setAge(i+1); user.setGender(1); users.add(user); } userService.saveOrUpdateBatch(users, 100); |
【本文地址】
今日新闻 |
推荐新闻 |